package com.binary.coder.gift.core;

import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.config.TemplateType;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.util.Collections;

/**
 * @author qinqin.luo.
 * @title
 * @time 2022/5/5.
 */
public class CodeGenerator {
    public static void main(String[] args) {

        FastAutoGenerator.create("jdbc:mysql://10.62.107.111:3306/imc_device?serverTimezone=Asia/Shanghai&useTimezone=true", "device", "device@123")
                .globalConfig(builder -> {
                    builder.author("shd").enableSwagger() // 设置作者
                            .fileOverride() // 覆盖已生成文件
                            .outputDir("F:\\generator");
                })
                .packageConfig(builder -> {
                    builder.parent("") // 设置父包名
                            .moduleName("") // 设置父包模块名
                            .pathInfo(Collections.singletonMap(OutputFile.xml,
                                    "F:\\generator\\code" )); // 设置mapperXml生成路径
                })
                .strategyConfig(builder -> {
                    builder.addInclude("status_breakdown_total")
                            // 设置需要生成的表名
                            .addTablePrefix("t_", "c_")
                    .entityBuilder()
                    .enableLombok()
                    .enableTableFieldAnnotation()

                    ; // 设置过滤表前缀
                })
                .templateConfig(builder -> {
//                    builder.disable(TemplateType.CONTROLLER,TemplateType.SERVICE,TemplateType.SERVICEIMPL,TemplateType.XML,TemplateType.MAPPER);
                    builder.disable(TemplateType.CONTROLLER,TemplateType.SERVICE,TemplateType.SERVICEIMPL);
                })
                .templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板
                .execute();
    }
}
